Skip to content

Conversation

@B-R-P
Copy link

@B-R-P B-R-P commented Jun 5, 2025

Implemented using Grok AI.
I didn't evaluate the code.
Please check is its correct implementation of DEBATER

B-R-P added 2 commits June 5, 2025 18:32
Implemented using Grok AI.
Evaluation of working pending...
- Implement Chain-of-Deliberation (CoD) with 8 appended [CoD] tokens for stepwise document embeddings.
- Add Self-Distillation (SD) with KL-divergence loss, including epsilon for numerical stability.
- Introduce dynamic max_length to prevent [CoD] token truncation, ensuring robust extraction.
- Add model compatibility checks for token addition and embedding resizing.
- Integrate FAISS-based retriever for efficient top-k retrieval with final CoD embeddings.
- Enhance logging and error handling for production reliability.

Addresses token placement, extraction robustness, model compatibility, and loss stability. Ready for E5 dataset and BEIR benchmark integration.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant